Skip to content

Reorder BEC reference values to match predicted BECs during training#1469

Merged
brucefan1983 merged 1 commit intobrucefan1983:masterfrom
MoseyQAQ:bec-reorder
Apr 27, 2026
Merged

Reorder BEC reference values to match predicted BECs during training#1469
brucefan1983 merged 1 commit intobrucefan1983:masterfrom
MoseyQAQ:bec-reorder

Conversation

@MoseyQAQ
Copy link
Copy Markdown
Contributor

Summary
Fix #1468
Specifically, in the qNEP, the BEC-related reference values were not reordered together with the structures. As a result, during non-full-batch training, the reference BEC values could become mismatched with the predicted BEC values in bec_train.out. This issue could also affect the BEC-related part of the training process itself.

Modification
Updated the reorder() logic in src/main_nep/structure.cu
Added reordering for the BEC-related reference data so that it remains consistent with the reordered structures

Validation
After the fix, the BEC fitting accuracy is improved. The potential energy surface fitting accuracy shows no obvious change.

image image

@brucefan1983 brucefan1983 merged commit 50cbe01 into brucefan1983:master Apr 27, 2026
2 of 3 checks passed
@MoseyQAQ MoseyQAQ deleted the bec-reorder branch April 27, 2026 14:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

qNEP predicts different BECs in training and prediction modes for the same train.xyz

2 participants